package com.chatdating.websocket.netty;

import com.chatdating.entity.pojo.TockenUserInfo;
import com.chatdating.entity.vo.LoginInVo;
import com.chatdating.redis.RedisComponet;
import com.chatdating.utils.EmptyUtil;
import com.chatdating.websocket.ChannelContextUtils;
import io.netty.channel.Channel;
import io.netty.channel.ChannelHandler;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.handler.codec.http.websocketx.TextWebSocketFrame;
import io.netty.handler.codec.http.websocketx.WebSocketServerProtocolHandler;
import io.netty.util.Attribute;
import io.netty.util.AttributeKey;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;


@ChannelHandler.Sharable     //设置通道共享
@Component("handlerWebSocket")
public class HandlerWebSocket extends SimpleChannelInboundHandler<TextWebSocketFrame> {
    private static final Logger logger = LoggerFactory.getLogger(HandlerWebSocket.class);

    @Resource
    private ChannelContextUtils channelContextUtils;

    @Resource
    private RedisComponet redisComponet;

    /**
     * 当通道就绪后会调用此方法，通常我们会在这里做一些初始化操作
     *
     * @param ctx
     * @throws Exception
     */
    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        logger.info("有新的连接加入。。。");
        logger.info("有新的连接加入。。。");
        logger.info("有新的连接加入。。。");
        logger.info("有新的连接加入。。。");
        logger.info("有新的连接加入。。。");
    }

    /**
     * 当通道不再活跃时（连接关闭）会调用此方法，我们可以在这里做一些清理工作
     *
     * @param ctx   （inactive） 不活跃的
     * @throws Exception
     */
    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        logger.info("有连接已经断开。。。");
        channelContextUtils.removeContext(ctx.channel());   //移除通道 并且更新用户最后离线时间
    }


    /**
     * 读就绪事件 当有消息可读时会调用此方法，我们可以在这里读取消息并处理。
     *
     * @param ctx
     * @param textWebSocketFrame
     * @throws Exception
     */
    @Override
    protected void channelRead0(ChannelHandlerContext ctx, TextWebSocketFrame textWebSocketFrame) throws Exception {
        //接收心跳
        Channel channel = ctx.channel();
        Attribute<String> attribute = channel.attr(AttributeKey.valueOf(channel.id().toString()));//获取通道属性
        String userId = attribute.get();
        redisComponet.saveUserHeartBeat(userId);  //保存用户心跳
    }


    //用于处理用户自定义的事件  当有用户事件触发时会调用此方法，例如连接超时，异常等。  （事件触发）
    //握手完成事件 （加上上面一共会调用两次）
    @Override
    public void userEventTriggered(ChannelHandlerContext ctx, Object evt) {
        if (evt instanceof WebSocketServerProtocolHandler.HandshakeComplete) {   //如果是 握手完成事件 判断tocken 是否正确
            WebSocketServerProtocolHandler.HandshakeComplete complete = (WebSocketServerProtocolHandler.HandshakeComplete) evt;
            String url = complete.requestUri();
            String token = getToken(url);  //根据 url 解析tocken  get请求方法
            if (token == null) {
                ctx.channel().close();
                return;
            }
             LoginInVo loginInVo = redisComponet.getTokenUserInfoDto(token); //根据tocken获取用户信息

            if (null == loginInVo) {
                ctx.channel().close();
                return;
            }
            /**
             * 连接成功之后 用户加入通道
             */
            channelContextUtils.addContext(loginInVo.getUsername().toString(), ctx.channel());

        }
    }


    //解析url 获取token
    private String getToken(String url) {
        if (EmptyUtil.isEmpty(url) || url.indexOf("?") == -1) { // 判断url是否为空  是否有 ？ 进行分割
            return null;
        }
        String[] queryParams = url.split("\\?");           // 用？进行分割  分出的结果是不是两部分
        if (queryParams.length < 2) {
            return url;
        }
        String[] params = queryParams[1].split("=");     // 用=进行分割  分出的结果是不是两部分
        if (params.length != 2) {
            return url;
        }
        return params[1];                                       //返回等号后面的数据 即 tocken
    }
}

